from operator import index
import sqlite3
import akshare as ak
import pandas as pd

"""
现在出现的问题是诱多跟进
1. 行业的成交量
2. 概念的成交量
3. 股票的成交量


大盘股无振幅
微盘股振幅大



"""

def fetch_a_share_history(symbols, start_date, end_date): 
    """
    获取A股历史数据并存入SQLite数据库
    Args:
        symbols: list of stock symbols (e.g. ['000001', '600000'])
        start_date: str 起始日期 (YYYYMMDD)
        end_date: str 结束日期 (YYYYMMDD)
    """
    import akshare as ak
    import sqlite3
    from tqdm import tqdm
    import traceback
    
    # 数据库连接
    conn = sqlite3.connect('stock_data.db')
    
    for symbol in tqdm(symbols, desc="正在获取股票数据"):
        try:
            # 获取历史数据
            hist = ak.stock_zh_a_hist(
                symbol=symbol,
                period="daily",
                start_date=start_date,
                end_date=end_date,
                adjust="hfq"
            )
            
            # 保存到数据库
            hist.to_sql(
                name="stock_zh_a_hist",
                con=conn,
                if_exists="append",
                index=False
            )
            
        except Exception as e:
            print(f"获取 {symbol} 数据失败: {str(e)}")
            traceback.print_exc()
    
    conn.close()

# 使用示例
fetch_a_share_history(
    symbols=['000001', '600000'],  # 股票代码列表
    start_date='20240101',         # 起始日期
    end_date='20500425'            # 结束日期
)


# 
